from llm import NovelLLM
from util import writeFile
from .base import BaseNode

class Step6Node(BaseNode):
    def run(self, state):
        prompt = "第六步：完整大纲\n当前大纲内容：{outline}\n请将每个段落扩展为一页详细内容".format(outline=state['step5_output'])
        result = NovelLLM().call_llm(prompt, caller_name="Step6")
        state['step6_output'] = result
        writeFile("output/step6.txt", result)
        print(f"Step6 完成完整大纲生成")
        return state